﻿// ------------------------------------------------------------------------
// 版权信息
// 版权归重庆虫儿飞科技有限公司所有。
// 所有权利保留。
// 官方网站：https://netokit.com
// 许可证信息
// Neto.Kit 项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。
// 许可证的完整文本可以在源代码树根目录中的 LICENSE-APACHE 和 LICENSE-MIT 文件中找到。
//
// 使用条款
// 使用本代码应遵守相关法律法规和许可证的要求。
//
// 免责声明
// 对于因使用本代码而产生的任何直接、间接、偶然、特殊或后果性损害，我们不承担任何责任。
//
// 其他重要信息
// Neto.Kit 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。
// 有关 Neto.Kit 项目的其他详细信息，请参阅位于源代码树根目录中的 COPYRIGHT 和 DISCLAIMER 文件。
//
// 更多信息
// 请访问 https://netokit.com 获取更多关于 Neto.Kit 项目的许可证和版权信息。
// ------------------------------------------------------------------------

using Microsoft.AspNetCore.Mvc.Filters;
using Microsoft.Extensions.DependencyInjection;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace Neto.Core;
[AttributeUsage(AttributeTargets.Method)]
public class PgSensitiveWordAttribute : Attribute, IAsyncActionFilter
{
    public char ReplaceChar { get; set; } = '*';
    public MatchMode MatchMode { get; set; } = MatchMode.Fuzzy;
    public int MinSeverity { get; set; } = 2;

    public async Task OnActionExecutionAsync(
        ActionExecutingContext context,
        ActionExecutionDelegate next)
    {
        var sensitiveService = context.HttpContext.RequestServices.GetService<SensitiveWordService>();

        // 处理所有输入参数
        foreach (var argument in context.ActionArguments.Values)
        {
            if (argument != null)
            {
                ProcessObject(argument, sensitiveService);
            }
        }

        await next();
    }
    /// <summary>
    /// 处理对象
    /// </summary>
    /// <param name="obj"></param>
    /// <param name="service"></param>
    private void ProcessObject(object obj, SensitiveWordService service)
    {
        if (obj == null) return;

        var type = obj.GetType();

        // 处理字符串属性
        foreach (var prop in type.GetProperties().Where(p => p.CanWrite && p.PropertyType == typeof(string)))
        {
            if (prop.GetValue(obj) is string value && !string.IsNullOrEmpty(value))
            {
                var newValue = service.ReplaceSensitiveWords(
                    value,
                    ReplaceChar,
                    MinSeverity,
                    MatchMode);

                prop.SetValue(obj, newValue);
            }
        }

        // 处理嵌套对象
        var complexProps = type.GetProperties()
            .Where(p => p.PropertyType.IsClass &&
                        p.PropertyType != typeof(string) &&
                        p.CanRead);

        foreach (var prop in complexProps)
        {
            var nestedObj = prop.GetValue(obj);
            if (nestedObj != null)
                ProcessObject(nestedObj, service);
        }
    }
}